import mmcv
import os
import tqdm

path_pic = '/media/hjh/workdir/0_Deep_Learning/TianChi/tile_round1_train_20201231/CZ_VOC2007/JPEGImages'
pic_list = os.listdir(path_pic)

shapes = []
for i in tqdm.tqdm(pic_list):
    # print(i)
    img_path = os.path.join(path_pic, i)
    img = mmcv.imread(img_path)
    h_w = img.shape[:2]
    if h_w not in shapes:
        shapes.append(h_w)

print(shapes)